home *** CD-ROM | disk | FTP | other *** search
/ Amiga Plus 2004 #2 / Amiga Plus CD - 2004 - No. 02.iso / AmigaPlus / Tools / mOS_Pegasos / cimon / cimon096src.lha / imon.c < prev    next >
Encoding:
C/C++ Source or Header  |  2002-04-05  |  4.4 KB  |  196 lines
  1. /* ----------------------------------------------------------------------
  2.    cimon - a UNIX command line client for the fli4l imon daemon.            
  3.  
  4.    Public domain, 2001-2002, Rene Herman <rene.herman@mail.com>         
  5. ---------------------------------------------------------------------- */
  6.  
  7. #include <signal.h>
  8. #include <string.h>           /* FD_ZERO expands to bzero() on MacOS X */
  9.  
  10. #include <unistd.h>
  11.  
  12. #include <sys/types.h>
  13. #include <sys/socket.h>
  14. #include <sys/time.h>
  15.  
  16. #ifdef __EMX__            /* EMX fails to declare select() in unistd.h */
  17. #include <sys/select.h>
  18. #endif
  19.  
  20. #include <netinet/in.h>
  21. #include <arpa/inet.h>
  22. #include <netdb.h>
  23.  
  24. #include "cimon.h"
  25.  
  26. /*--- globals */
  27.  
  28. unsigned short port = IMON_PORT; /* TCP port number             */      
  29. const char    *password;         /* imond password              */
  30.  
  31. /*--- filescopes */
  32.  
  33. static int  sock;                /* socket                       */
  34. static int  empty = 1;           /* communications channel empty */
  35.  
  36. #define     BUFF_SIZE 4096       /* 0 < BUFF_SIZE <= SSIZE_MAX  */
  37. static char buff[BUFF_SIZE];     /* command/reply buffer        */
  38. static int  bufc = 0;            /* buffer counter              */
  39.  
  40. /*--- private functions */
  41.  
  42. static void buff_send(void)
  43. {
  44.     if (write(sock, buff, bufc) != bufc)
  45.         exit_errno("Write error");
  46.     empty = bufc = 0;
  47. }
  48.  
  49. static void buff_putc(const char c)
  50. {
  51.     if (bufc == BUFF_SIZE)
  52.         buff_send();
  53.     buff[bufc++] = c;
  54. }
  55.  
  56. static void buff_puts(const char *s)
  57. {
  58.     while (*s)
  59.         buff_putc(*s++);
  60. }
  61.  
  62. static void send_command(void)
  63. {
  64.     buff_putc('\r');
  65.     buff_putc('\n');
  66.     buff_send();
  67. }
  68.  
  69. static void send_quit(void)
  70. {
  71.     buff_puts("quit");
  72.     send_command();
  73. } 
  74.  
  75. #define REPLY_OK  1
  76. #define REPLY_ERR 2
  77.  
  78. static void get_reply(void)
  79. {
  80.     static char *s;
  81.     static char  c = '\n';
  82.  
  83.     fd_set         sockset;
  84.     struct timeval timeout;
  85.     int            i;
  86.     
  87.     /* 10 second timeout */
  88.     FD_ZERO(&sockset);
  89.     FD_SET(sock, &sockset);      
  90.     timeout.tv_sec = 10;
  91.     timeout.tv_usec = 0;
  92.     
  93.     if ((i = select(sock + 1, &sockset, NULL, NULL, &timeout)) < 0)
  94.         exit_errno("Internal error (select)");
  95.     else if (!i)
  96.         exit_error("Timeout waiting for reply from remote");
  97.     
  98.     if ((i = read(sock, buff, BUFF_SIZE)) < 0)
  99.         exit_errno("Read error");
  100.     else if (!i)
  101.         exit_error("Connection closed by remote");
  102.     
  103.     /* scan for (\n terminated) line starting with OK or ERR */
  104.     do {
  105.         if (c == '\n')
  106.             s = buff[bufc] == 'O' ? "K" : buff[bufc] == 'E' ? "RR" : NULL;
  107.         else if (s && *s && buff[bufc] != *s++)
  108.             s = NULL;
  109.         c = buff[bufc++];
  110.     } while (--i);
  111.  
  112.     /* "empty" doubles as OK/ERR indicator */    
  113.     empty = s && !*s && c == '\n' ?
  114.         (*(s - 1) == 'K' ? REPLY_OK : REPLY_ERR) : 0;
  115. }
  116.  
  117. static void send_command_get_reply()
  118. {
  119.     send_command();
  120.     do {
  121.         get_reply();
  122.         write(STDOUT_FILENO, buff, bufc);
  123.         bufc = 0;
  124.     } while (!empty);
  125. }
  126.  
  127. static void check_password(void)
  128. {
  129.     buff_puts("pass ");
  130.     buff_puts(password);
  131.     send_command();
  132.     do {
  133.         get_reply();
  134.         bufc = 0;
  135.     } while (!empty);
  136.     if (empty != REPLY_OK) {
  137.         send_quit();
  138.         exit_error("Password rejected by remote");
  139.     }
  140. }
  141.  
  142. static void flush()
  143. {
  144.     if (bufc || !empty)
  145.         send_command_get_reply();
  146. }
  147.  
  148. /*--- global functions */
  149.  
  150. void imon_open(const char * const hostname)
  151. {
  152.     struct sockaddr_in name;
  153.     struct hostent    *host;
  154.  
  155.     name.sin_family = AF_INET;
  156.     name.sin_port   = htons(port);
  157.     if ((name.sin_addr.s_addr = inet_addr(hostname)) == INADDR_NONE) {
  158.         if (!(host = gethostbyname(hostname)))
  159.             exit_error("Can't find `%s': %s", hostname, herrorstr(h_errno));
  160.         name.sin_addr = *(struct in_addr *)host->h_addr;
  161.     }
  162.  
  163.     if ((sock = socket(PF_INET, SOCK_STREAM, 0)) < 0)
  164.         exit_errno("Error creating socket");
  165.     if (connect(sock, (struct sockaddr *)&name, sizeof name))
  166.         exit_errno(hostname);
  167.  
  168.     if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
  169.         exit_errno("Internal error (signal)"); 
  170.  
  171.     if (password)
  172.         check_password();
  173. }
  174.  
  175. void imon_putc(const char c)
  176. {
  177.     if (c == '\n')
  178.         flush();
  179.     else
  180.         buff_putc(c);
  181. }
  182.  
  183. void imon_puts(const char *s)
  184. {
  185.     while (*s)
  186.         imon_putc(*s++);
  187. }
  188.  
  189. void imon_quit(void)
  190. {
  191.     flush();
  192.     send_quit();
  193.     if (close(sock)) 
  194.         exit_errno("Can't close connection");
  195. }
  196.